from django.core.cache import cache
from django.http import JsonResponse
from django.shortcuts import render
from django.views import View
from apps.areas.models import Area
from utils.response_code import RETCODE

import logging

logger = logging.getLogger('django')

class AreasView(View):
    """省市区数据"""

    def get(self, request):
        """提供省市区数据"""
        area_id = request.GET.get('area_id')

        if not area_id:
            # 读取省份缓存数据
            province_list = cache.get('province_list')

            if not province_list:
                # 提供省份数据
                try:
                    # 查询省份数据
                    province_model_list = Area.objects.filter(parent__isnull=True)

                    # 序列化省级数据
                    province_list = []
                    for province_model in province_model_list:
                        province_list.append({
                            'id': province_model.id,
                            'name': province_model.name})
                except Exception as e:
                    logger.error(e)
                    return JsonResponse({'code': RETCODE.DBERR, 'errmsg': '省份数据错误'})

                # 存储省份缓存数据
                cache.set('province_list', province_list, 24*3600)

            # 响应省份数据
            return JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'provinces': province_list})
        else:
            # 读取市或区缓存数据
            sub_list = cache.get('sub_area_' + area_id)

            if not sub_list:
                # 提供市或区数据
                try:
                    # 根据id查询市或区
                    area_model = Area.objects.filter(parent_id=area_id)

                    # 序列化市或区数据
                    sub_list = []
                    for sub_model in area_model:


                        sub_list.append({
                            'id': sub_model.id,
                            'name': sub_model.name})

                except Exception as e:
                    logger.error(e)
                    return JsonResponse({'code': RETCODE.DBERR, 'errmsg': '城市或区数据错误'})

                # 储存市或区缓存数据
                cache.set('sub_area_' + area_id, sub_list, 24*3600)

            # 响应市或区数据
            return JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'sub_list': sub_list})